package com.hardy.leecode;

import com.hardy.leecode.model.ListNode;

/**
 * Author: Hardy
 * Date:   2020/6/26
 * Description:
 * - 两数相加
 * 给出两个 非空 的链表用来表示两个非负的整数。其中，它们各自的位数是按照 逆序 的方式存储的，并且它们的每个节点只能存储 一位 数字。
 * 如果，我们将这两个数相加起来，则会返回一个新的链表来表示它们的和。
 * 您可以假设除了数字 0 之外，这两个数都不会以 0 开头。
 * <p>
 * 示例：
 * <p>
 * 输入：(2 -> 4 -> 3) + (5 -> 6 -> 4)
 * 输出：7 -> 0 -> 8
 * 原因：342 + 465 = 807
 * <p>
 * 链接：https://leetcode-cn.com/problems/add-two-numbers/
 **/
public class Que2 {

    public static void main(String[] args) {

        ListNode l1 = ListNode.gen(new int[]{2, 4, 3});
        ListNode l2 = ListNode.gen(new int[]{5, 6, 4});

        System.out.println(new Solution().addTwoNumbers(l1, l2));
    }

    static class Solution {
        public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
            int temp = 0;
            ListNode head = new ListNode(0);
            ListNode node = head;

            while (!(l1 == null && l2 == null && temp == 0)) {
                // 只有进位
                if (l1 == null && l2 == null) {
                    node.next = new ListNode(temp);
                    break;
                }
                // 剩下单链表
                if (l1 == null || l2 == null) {
                    // 链表拼接
                    node.next = l1 == null ? l2 : l1;
                    // 进位拼接
                    while (temp == 1) {
                        ListNode next = node.next;
                        if (next == null) {
                            node.next = new ListNode(temp);
                            temp = 0;
                        } else {
                            int val = next.val + temp;
                            if (val >= 10) {
                                val = val - 10;
                            } else {
                                temp = 0;
                            }
                            next.val = val;
                            node = next;
                        }
                    }
                    break;
                }
                // 两个都存在
                int val = l1.val + l2.val + temp;
                if (val >= 10) {
                    val = val - 10;
                    temp = 1;
                } else {
                    temp = 0;
                }
                node.next = new ListNode(val);

                l1 = l1.next;
                l2 = l2.next;
                node = node.next;
            }

            return head.next;
        }
    }
}
